package config

import (
	"database/sql"
	"log"

	_ "modernc.org/sqlite"
)

// InitDB 初始化数据库连接
func InitDB() *sql.DB {
	db, err := sql.Open("sqlite", "./local-sqlLite.db")
	if err != nil {
		log.Fatal(err)
	}

	// 测试数据库连接
	if err = db.Ping(); err != nil {
		log.Fatal(err)
	}

	return db
}